config_full = {
    'sr': 88200,
    'frame_rate': 60,
    'hidden_dim': 1024,
    'n_layers': 8, 
    'num_heads': 16, 
    'dropout': 0.0,
    'output_dim': 68, # if you trained your own, this should also be 61
    'input_dim': 256,
    'frame_size': 128, 
    'use_half_precision': False,
    'model_path': 'utils/model/model-full.pth'
}
config_512_4_8 = {
    'sr': 88200,
    'frame_rate': 60,
    'hidden_dim': 512,
    'n_layers': 4, 
    'num_heads': 8, 
    'dropout': 0.3,
    'output_dim': 61, # if you trained your own, this should also be 61
    'input_dim': 256,
    'frame_size': 128, 
    'use_half_precision': False,
    'model_path': 'utils/model/model-512-4-8.pth'
}
config_small= {
    'sr': 88200,
    'frame_rate': 60,
    'hidden_dim': 128,
    'n_layers': 8, 
    'num_heads': 8, 
    'dropout': 0.3,
    'output_dim': 61, # if you trained your own, this should also be 61
    'input_dim': 256,
    'frame_size': 128, 
    'use_half_precision': False,
    'model_path': 'utils/model/model-small.pth'
}
config_tiny = {
    'sr': 88200,
    'frame_rate': 60,
    'hidden_dim': 128,
    'n_layers': 4, 
    'num_heads': 4, 
    'dropout': 0.3,
    'output_dim': 61, # if you trained your own, this should also be 61
    'input_dim': 256,
    'frame_size': 128, 
    'use_half_precision': False,
    'model_path': 'utils/model/model-tiny.pth'
}

config = config_tiny